Training-Free On-Policy Distillation
免训练的 On-Policy 蒸馏
🧭 无训练模型蒸馏与多专家聚合
University of Illinois at Urbana-Champaign · Tsinghua University
提出免训练 on-policy 蒸馏,仅用前向传播和最小二乘逐层合并多个专家模型,无需梯度更新
📌 工作参考价值(4 点)
- 多领域专家模型合并:若业务中有多个针对不同场景(如推荐、搜索、广告)微调的模型,可借鉴 TF-OPD 通过前向计算几百个 rollout 提取输入二阶矩,逐层最小二乘融合,无需联合训练,节省算力和迭代时间。
- 模型融合的精度提升:相比简单 task arithmetic 权重平均,逐层最小二乘匹配专家输出分布能保留更多能力;在推荐模型做多目标融合或跨域迁移时可引入此方法提升融合质量。
- 数据筛选策略:论文发现去除重复循环的生成 token 显著提升融合效果(AIME25 从 task arithmetic 水平到专家水平),提示在蒸馏或融合数据准备中应重点清洗低质量重复内容,避免噪声影响层权重估计。
- 工程部署:融合只需每专家数 GPU 小时的前向计算,适合快速将多个专家模型合并为通用模型上线,尤其适用于需要频繁更新多领域能力的 Agent 或推荐系统。